package com.blog.util;

import com.auth0.jwt.JWT;
import com.auth0.jwt.algorithms.Algorithm;
import com.blog.properties.JwtProperties;
import com.auth0.jwt.interfaces.DecodedJWT;
import com.auth0.jwt.interfaces.Claim;

import java.util.Date;
import java.util.Map;
import java.util.HashMap;

public class JwtUtil {

    /**
     *生成jwt
     * 使用Hs256算法， 私钥使用固定个人密钥
     *
     * @param secreKey  jwt密钥
     * @param time      jwt过期时间
     * @param claims    设置的信息
     * @return
     */
    public static String  createJWT(String secreKey, long time, Map<String, Object> claims){
        return JWT.create() // 生成Buider，进行构造
                .withClaim("claims", claims) // 要转换的数据
                .withExpiresAt(new Date(System.currentTimeMillis() + time)) // 指定过去多久时间过期
                .sign(Algorithm.HMAC256(secreKey)); // 指定使用Hs256算法， 私钥使用固定个人密钥
     }

    /**
     * Token解密
     *
     * @param token     加密后的token
     * @return
     */
     public static Map<String, Object> parseToken(String token){
        DecodedJWT decodedJWT = JWT.require(Algorithm.HMAC256(JwtProperties.secretKey))
                .build()
                .verify(token);
        
        Map<String, Object> claims = decodedJWT.getClaim("claims").asMap();
        Map<String, Object> resultMap = new HashMap<>();
        
        claims.forEach((k, v) -> {
            switch (k) {
                case "id":
                    resultMap.put(k, Long.valueOf(v.toString())); // 将Object转为String再转为Long
                    break;
                case "role":
                case "username":
                case "adminName":
                    resultMap.put(k, v.toString());
                    break;
                default:
                    resultMap.put(k, v); // 直接存入,保持原有类型
            }
        });
        
        return resultMap;
     }
}
